//
//  135. 分发糖果.swift
//  手撕代码
//
//  Created by xiaoZuDeMeng on 2024/2/1.
//

import Foundation

//n 个孩子站成一排。给你一个整数数组 ratings 表示每个孩子的评分。
//
//你需要按照以下要求，给这些孩子分发糖果：
//
//每个孩子至少分配到 1 个糖果。
//相邻两个孩子评分更高的孩子会获得更多的糖果。
//请你给每个孩子分发糖果，计算并返回需要准备的 最少糖果数目 。
//
//
//
//示例 1：
//
//输入：ratings = [1,0,2]
//输出：5
//解释：你可以分别给第一个、第二个、第三个孩子分发 2、1、2 颗糖果。
//示例 2：
//
//输入：ratings = [1,2,2]
//输出：4
//解释：你可以分别给第一个、第二个、第三个孩子分发 1、2、1 颗糖果。
//     第三个孩子只得到 1 颗糖果，这满足题面中的两个条件。
//
//
//提示：
//
//n == ratings.length
//1 <= n <= 2 * 104
//0 <= ratings[i] <= 2 * 104

func 分发糖果(_ ratings: [Int]) -> Int {
    var lefts = Array(repeating: 1, count: ratings.count)
    for index in 0..<(ratings.count - 1) {
        if ratings[index] < ratings[index + 1] {
            lefts[index + 1] = lefts[index] + 1
        }
    }
    var result = lefts[ratings.count - 1]
    var rights = Array(repeating: 1, count: ratings.count)
    for index in (0..<(ratings.count - 1)).reversed() {
        if ratings[index] > ratings[index + 1] {
            rights[index] = rights[index + 1] + 1
        }
        result += max(lefts[index], rights[index])
    }
    return result
}
